/*
 * Copyright (C) 2005-2011 MaNGOS <http://www.getmangos.com/>
 *
 * Copyright (C) 2008-2011 Trinity <http://www.trinitycore.org/>
 *
 * Copyright (C) 2010-2011 ProjectSkyfire <http://www.projectskyfire.org/>
 * 
 * Copyright (C) 2011 ArkCORE <http://www.arkania.net/>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

/** \file
 \ingroup realmd
 */

#ifndef __REALMSOCKET_H__
#define __REALMSOCKET_H__

#include <ace/Synch_Traits.h>
#include <ace/Svc_Handler.h>
#include <ace/SOCK_Stream.h>
#include <ace/Message_Block.h>
#include <ace/Basic_Types.h>

class RealmSocket: public ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_NULL_SYNCH> {
private:
	typedef ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_NULL_SYNCH> Base;

public:
	class Session {
	public:
		Session(void);
		virtual ~Session(void);

		virtual void OnRead(void) = 0;
		virtual void OnAccept(void) = 0;
		virtual void OnClose(void) = 0;
	};

	RealmSocket(void);
	virtual ~RealmSocket(void);

	size_t recv_len(void) const;
	bool recv_soft(char *buf, size_t len);
	bool recv(char *buf, size_t len);
	void recv_skip(size_t len);

	bool send(const char *buf, size_t len);

	const std::string& get_remote_address(void) const;

	virtual int open(void *);

	virtual int close(int);

	virtual int handle_input(ACE_HANDLE = ACE_INVALID_HANDLE);
	virtual int handle_output(ACE_HANDLE = ACE_INVALID_HANDLE);

	virtual int handle_close(ACE_HANDLE = ACE_INVALID_HANDLE, ACE_Reactor_Mask =
			ACE_Event_Handler::ALL_EVENTS_MASK);

	void set_session(Session* session);

private:
	ssize_t noblk_send(ACE_Message_Block &message_block);

	ACE_Message_Block input_buffer_;
	Session* session_;
	std::string remote_address_;
};

#endif /* __REALMSOCKET_H__ */
